#include "UDPServe.hpp"
#include "Route.hpp"
#include "ThreadPool.hpp"

using namespace LogModule;
using namespace ThreadPoolModule;
std::string defaulthandle(const std::string& message)
{
    std::string hello="hello";
    hello+=message;
    return hello;
}
using task_t = std::function<void()>;
int main(int argc,char*argv[])
{
    // if(argc!=3)//一会改一下
    if(argc!=2)
    {
        std::cerr<<"Usage:"<<argv[0]<<" Port"<<std::endl;
        return 1;
    }
    
    // std::string ip=argv[1];
    uint16_t port=std::stoi(argv[1]);
    Enable_Console_Sync_Strategy;
    
    //1.Route function
    // Dict* dict=new Dict();
    // dict->LoadDictionary();
    Route r;

    auto tp = ThreadPool<task_t>::GetInstance();

    std::unique_ptr<UdpServe> m = std::make_unique<UdpServe>(port, [&r, &tp](int sockfd, const std::string &message, InetAddr peer){
        task_t t = std::bind(&Route::MessageRoute, &r, sockfd, message, peer);
        tp->EnterQueue(t);
    });
    // std::unique_ptr<UdpServe> m=std::make_unique<UdpServe>(port,[&r](int sockfd,const std::string& message,InetAddr& peer){
    //     r.MessageRoute(sockfd,message,peer);
    // });
    m->Init();
    m->Start();
    return 0;    
}